package com.sh.rbac.handler;

import com.sh.rbac.exception.CustomException;
import com.sh.rbac.result.Result;
import com.sh.rbac.result.ResultCodeEnum;
import lombok.extern.slf4j.Slf4j;
import org.springframework.security.access.AccessDeniedException;
import org.springframework.validation.BindingResult;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import java.util.Objects;

/**
 * 全局异常处理
 *
 * @author wangbing
 * @since 2024-03-14 17:04
 */

@Slf4j
@RestControllerAdvice
public class GlobalExceptionHandler {


    /**
     * 全局异常处理
     *
     * @param e 异常
     * @return 结果
     */
    @ExceptionHandler(Exception.class)
    public Result<Object> error(Exception e) {
        log.error("全局异常处理-->{}", e.getMessage(), e);
        return Result.failure(e.getMessage());
    }

    /**
     * 权限不足异常
     *
     * @param e 异常
     * @return 结果
     */
    @ExceptionHandler(AccessDeniedException.class)
    public Result<Object> error(AccessDeniedException e) {
        log.error("权限不足异常处理-->{}", e.getMessage(), e);
        return Result.failure(ResultCodeEnum.AUTH_NOT_PERMISSION);
    }

    /**
     * 参数校验异常
     *
     * @param e 异常
     * @return 结果
     */
    @ExceptionHandler(MethodArgumentNotValidException.class)
    public Result<Object> error(MethodArgumentNotValidException e) {
        log.error("参数异常处理-->{}", e.getMessage(), e);
        // 获取校验结果
        BindingResult bindingResult = e.getBindingResult();
        // 获取校验错误信息
        return Result.failure(Objects.requireNonNull(bindingResult.getFieldError()).getDefaultMessage());
    }


    /**
     * 自定义异常
     *
     * @param e 异常
     * @return 结果
     */
    @ExceptionHandler(CustomException.class)
    public Result<Object> error(CustomException e) {
        log.error("自定义异常处理-->{}", e.getMessage(), e);
        return Result.failure(e.getMessage());
    }
}
